How to call a python function based on a pandas column value where the function called returns a dat